# imports
from matplotlib.colors import Colormap

# unit tests


class TestColormapWithExtremes:
    # Basic Test Cases

    def test_returns_new_instance(self):
        # Should return a new Colormap instance, not the same object
        cm = Colormap("test")
        codeflash_output = cm.with_extremes(bad=(1, 0, 0, 1))
        cm2 = codeflash_output  # 7.51μs -> 7.11μs (5.60% faster)

# imports
import pytest
from matplotlib.colors import Colormap

# unit tests

# ---- Basic Test Cases ----


def test_with_extremes_returns_copy():
    # Test that with_extremes returns a different object, not self
    cmap = Colormap("test")
    codeflash_output = cmap.with_extremes(bad=(1, 0, 0, 1))
    cmap2 = codeflash_output  # 9.30μs -> 7.84μs (18.6% faster)


def test_with_extremes_invalid_color_tuple_length():
    # Should raise if color tuple is not length 4
    cmap = Colormap("test")
    with pytest.raises(ValueError):
        cmap.with_extremes(bad=(1, 2, 3))  # 6.63μs -> 5.87μs (12.9% faster)


def test_with_extremes_invalid_color_tuple_values():
    # Should raise if color tuple contains values out of range
    cmap = Colormap("test")
    with pytest.raises(ValueError):
        cmap.with_extremes(bad=(1.5, -0.1, 0.5, 0.5))  # 8.02μs -> 7.30μs (9.89% faster)


# Patch set_extremes to check for tuple length and range
def patched_set_extremes(self, bad=None, under=None, over=None):
    for color, name in zip((bad, under, over), ("bad", "under", "over")):
        if color is not None:
            if not (isinstance(color, tuple) and len(color) == 4):
                raise ValueError(f"{name} color must be a tuple of length 4")
            if not all(isinstance(c, (int, float)) and 0 <= c <= 1 for c in color):
                raise ValueError(f"{name} color values must be in [0, 1]")
    if bad is not None:
        self._rgba_bad = bad
    if under is not None:
        self._rgba_under = under
    if over is not None:
        self._rgba_over = over
    if self._isinit:
        self._init()


Colormap.set_extremes = patched_set_extremes